from torch.utils.data import DataLoader

# 假设 MyDataset 是你自定义的数据集类
train_dataset = MyDataset(root='path/to/train/data', transform=my_transforms)
train_loader = DataLoader(train_dataset, batch_size=64, shuffle=True)
